package com.huanzhidadi.single.proxy;

import org.springframework.cglib.proxy.*;

import java.lang.reflect.Method;

/**
 * Cglib动态代理，基于ASM机制实现
 */
public class CglibProxy2Test {
    public static void main(String[] args) {
        ProxyInterceptor2 proxyInterceptor2 = new ProxyInterceptor2(new TargetObject2());
        TargetObject2 targetObject = (TargetObject2) proxyInterceptor2.getProxy();
        targetObject.say();
        targetObject.work();
    }
}

// 被代理类
class TargetObject2 {
    public void say() {
        System.out.println("被代理类 say");
    }

    public void work() {
        System.out.println("被代理类 work");
    }
}

/**
 * 继承 MethodInterceptor，用于方法的拦截回调
 */
class ProxyInterceptor2 implements MethodInterceptor {

    // 字节码增强器,它可以方便的对你想要处理的类进行扩展
    // 动态的创建给定类的子类并且拦截代理类的所有的方法
    private Enhancer enhancer = new Enhancer();

    // 目标类
    private Object bean;

    public ProxyInterceptor2(Object bean) {
        this.bean = bean;
    }

    public Object getProxy(){
        // 设置需要创建代理类的类
        enhancer.setSuperclass(bean.getClass());
        // 设置拦截器
        enhancer.setCallback(this);
        // 通过字节码技术动态创建代理类实例
        return enhancer.create();
    }

    /**
     * @param object      表示要进行增强的对象
     * @param method      表示拦截的方法
     * @param args     数组表示参数列表，基本数据类型需要传入其包装类型，如int-->Integer、long-Long、double-->Double
     * @param methodProxy 表示对方法的代理，invokeSuper方法表示对被代理对象方法的调用
     * @return 执行结果
     * @throws Throwable
     */
    @Override
    public Object intercept(Object object, Method method, Object[] args, MethodProxy methodProxy) throws Throwable {
        if ("say".equals(method.getName())) {
            System.out.println("代理类 say hello");
        } else if ("work".equals(method.getName())) {
            System.out.println("代理类 work");
        }
        // 注意这里是调用 invokeSuper 而不是 invoke，否则死循环，methodProxy.invokesuper执行的是原始类的方法，method.invoke执行的是子类的方法
        Object result = methodProxy.invokeSuper(object, args);
        return result;
    }
}
